//
// connection.cpp
// ~~~~~~~~~~~~~~
//
// Copyright (c) 2003-2012 Christopher M. Kohlhoff (chris at kohlhoff dot com)
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//

#include "connection.hpp"
#include <vector>
#include <boost/bind.hpp>
#include "request_handler.hpp"

namespace http {
namespace server3 {

connection::connection(boost::asio::io_service& io_service, request_handler& handler, Log& log_name)
	: strand_(io_service), socket_(io_service), request_handler_(handler), server_log_(log_name)
{
}

boost::asio::ip::tcp::socket& connection::socket()
{
	return socket_;
}

void connection::start()
{
	socket_.async_read_some(boost::asio::buffer(buffer_),
		strand_.wrap(boost::bind(&connection::handle_read, shared_from_this(),
			boost::asio::placeholders::error,
			boost::asio::placeholders::bytes_transferred)));
}

void connection::handle_read(const boost::system::error_code& e, std::size_t bytes_transferred)
{
	if (!e)
	{
		boost::tribool result;
		boost::tie(result, boost::tuples::ignore) = request_parser_.parse(request_, buffer_.data(), buffer_.data() + bytes_transferred);

		if (result)
		{
			server_log_.log_for_request(socket_.remote_endpoint().address().to_string(), socket_.remote_endpoint().port(), request_.uri);
			request_handler_.handle_request(request_, reply_);
			if (reply_.status == reply::bad_request)
				server_log_.log_for_error(socket_.remote_endpoint().address().to_string(), socket_.remote_endpoint().port(), request_.uri, "bad request");
			else if (reply_.status == reply::not_found)
				server_log_.log_for_error(socket_.remote_endpoint().address().to_string(), socket_.remote_endpoint().port(), request_.uri, "not found");
			boost::asio::async_write(socket_, reply_.to_buffers(),
				strand_.wrap(boost::bind(&connection::handle_write, shared_from_this(),
					boost::asio::placeholders::error)));
		}
		else if (!result)
		{
			reply_ = reply::stock_reply(reply::bad_request);
			boost::asio::async_write(socket_, reply_.to_buffers(),
				strand_.wrap(boost::bind(&connection::handle_write, shared_from_this(),
					boost::asio::placeholders::error)));
		}
		else
		{
			socket_.async_read_some(boost::asio::buffer(buffer_),
				strand_.wrap(boost::bind(&connection::handle_read, shared_from_this(),
					boost::asio::placeholders::error,
					boost::asio::placeholders::bytes_transferred)));
		}
	}

	// If an error occurs then no new asynchronous operations are started. This
	// means that all shared_ptr references to the connection object will
	// disappear and the object will be destroyed automatically after this
	// handler returns. The connection class's destructor closes the socket.
}

void connection::handle_write(const boost::system::error_code& e)
{
	if (!e)
	{
		if (reply_.status == reply::ok)
			server_log_.log_for_send(socket_.remote_endpoint().address().to_string(), socket_.remote_endpoint().port(), request_.uri, reply_.content.size(), reply_timer.elapsed() * 1000);
		// Initiate graceful connection closure.
		boost::system::error_code ignored_ec;
		socket_.shutdown(boost::asio::ip::tcp::socket::shutdown_both, ignored_ec);
	}

	// No new asynchronous operations are started. This means that all shared_ptr
	// references to the connection object will disappear and the object will be
	// destroyed automatically after this handler returns. The connection class's
	// destructor closes the socket.
}

} // namespace server3
} // namespace http
